#!/usr/bin/env python
# -*- coding: utf-8 -*-

import random
import unittest

class TestSequenceFunctions(unittest.TestCase):
    def setUp(self):
        # Creamos una lista con 10 elementos en la que el elemento 
        # número "i" es un entero de valor "i":
        self.seq = range(10)

    def test_shuffle(self):
        # Nos aseguramos de que la secuencia aleatoria no pierde 
        # ningún elemento:
        random.shuffle(self.seq)
        self.seq.sort()
        self.assertEqual(self.seq, range(10))

        # Debería lanzar una excepción debido a una secuencia 
        # inmutable:
        self.assertRaises(TypeError, random.shuffle, (1,2,3))

    def test_choice(self):
        element = random.choice(self.seq)
        self.assertTrue(element in self.seq)

    def test_sample(self):
        with self.assertRaises(ValueError):
            random.sample(self.seq, 20)
        for element in random.sample(self.seq, 5):
            self.assertTrue(element in self.seq)

if __name__ == '__main__':
    unittest.main()





